# coding=utf-8
# author=haishan09@gmail.com
from django.contrib import admin
from dimension.models import Widget, Provider, Dimension
from dimension.forms import ProviderForm, DimensionForm


class WidgetAdmin(admin.ModelAdmin):

    model = Widget


class ProviderAdmin(admin.ModelAdmin):

    model = Provider

    def get_form(self, request, obj=None, **kwargs):
        return ProviderForm 

class DimensionAdmin(admin.ModelAdmin):

    model = Dimension

    def get_form(self, request, obj=None, **kwargs):
        return DimensionForm 


admin.site.register(Widget, WidgetAdmin)
admin.site.register(Provider, ProviderAdmin)
admin.site.register(Dimension, DimensionAdmin)

